package com.client.controller;
import java.awt.Dimension;
import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.imageio.ImageIO;
public class screenTest {
    private String fileName; //文件的前缀
    private String defaultName = "GuiCamera";
    static int serialNum=0;
    private String imageFormat; //图像文件的格式
    private String defaultImageFormat="png";
    Dimension d = Toolkit.getDefaultToolkit().getScreenSize();

    /****************************************************************
    * 默认的文件前缀为GuiCamera，文件格式为PNG格式
    * The default construct will use the default
    * Image file surname "GuiCamera",
    * and default image format "png"
    ****************************************************************/
    public screenTest(){
        fileName = defaultName;
        imageFormat=defaultImageFormat;

    }
    /****************************************************************
      * @param s the surname of the snapshot file
      * @param format the format of the image file,
      * it can be "jpg" or "png"
      * 本构造支持JPG和PNG文件的存储
      ****************************************************************/
    public screenTest(String s,String format){
        fileName = s;
        imageFormat=format;
    }
    /****************************************************************
      * 对屏幕进行拍照
      * snapShot the Gui once
      ****************************************************************/
    public int snapShot() {
        int res=-1;
        try {
            //拷贝屏幕到一个BufferedImage对象screenshot
            BufferedImage screenshot = (new Robot()).createScreenCapture(new
                    Rectangle(0, 0, (int) d.getWidth(), (int) d.getHeight()));
            serialNum++;
            res=serialNum;
            //根据文件前缀变量和文件格式变量，自动生成文件名
            String name = fileName + String.valueOf(serialNum) + "." + imageFormat;
            File f = new File(name);
            System.out.print("Save File " + name);
//将screenshot对象写入图像文件
            ImageIO.write(screenshot, imageFormat, f);
            System.out.print("..Finished!\n");
        }
        catch (Exception ex) {
            System.out.println(ex);
        }
        return res;
    }
}
